/*
 * TCP服务器.c
 * 
 * 
 */

#include <netdb.h>
#include <sys/socket.h>
#include <unistd.h>
#include <string.h>
#include <stdio.h>
#include <stdlib.h>
#include <memory.h>
#include <signal.h>
#include <arpa/inet.h>

#include <wiringPi.h>
#define LED 7

int sockfd;
char flag=1;

void GPIOSetup(void)
{
	wiringPiSetup();
	pinMode(LED,OUTPUT);
	digitalWrite(LED,0);
}

//登记信号处理函数，使程序接收ctrl+C信号时关闭服务器
void sig_handler(int signo)
{
	if(signo==SIGINT)
	{
		printf("server close\n");
		/*步骤6：关闭客户端的socket*/
		close(sockfd);
		exit(1);
	}
}

/*输出连接上来的客户端相关信息*/
void out_addr(struct sockaddr_in *clientaddr)
{
	//将端口号从网络字节序转换成主机字节序
	int port = ntohs(clientaddr->sin_port);
	char ip[16];
	memset(ip,0,sizeof(ip));
	//将ip地址从网络字节序转换成点分十进制
	inet_ntop(AF_INET,
					&clientaddr->sin_addr.s_addr,
					ip, sizeof(ip));
	printf("client: %s(%d) connected\n",ip,port);
}

/*执行对客户端操作*/
void do_service(int fd)
{
	char command=0;
	
	while(1){
		read(fd,&command,sizeof(char));
		printf("%c",command);
		
		//具体操作
		switch(command){
			case 'E':
				if(flag){
					printf("打开小灯\n");
					digitalWrite(LED,1);
				}
				else{
					printf("关闭小灯\n");
					digitalWrite(LED,0);
				}
				flag ^= 1;
				break;
				
			default:break;
		}
		
		if(write(fd, &command, 1) != 1)
		{perror("write error");}
	}
	
}

/*主函数*/
int main(int argc, char **argv)
{
	
	if(argc<2)
	{
		printf("usage:%s #port\n",argv[0]); //传递端口参数
		exit(1);
	}
	
	GPIOSetup();
	
	/*处理ctl+c信号，关闭服务器*/
	if(signal(SIGINT, sig_handler)==SIG_ERR) 
	{
		perror("signal sigint error");
		exit(1);
	}
	
	/*步骤1：创建socket
	 * 注：socket创建在内核中，是一个结构体。
	 * AF_INET：IPv4
	 * SOCK_STREAM:tcp协议
	 * */
	sockfd = socket(AF_INET, SOCK_STREAM, 0);
	
	/*步骤2：将socket和地址(包括ip、port)进行绑定*/
	struct sockaddr_in serveraddr;
	memset(&serveraddr,0,sizeof(serveraddr));
	//往地址中填入ip、port、internet地址族类型
	serveraddr.sin_family = AF_INET; //IPv4
	serveraddr.sin_port = htons(atoi(argv[1])); //port
	serveraddr.sin_addr.s_addr = INADDR_ANY; //INADDR_ANY响应所有客户端请求
	if(bind(sockfd, (struct sockaddr*)&serveraddr,
				sizeof(serveraddr))<0)
		{perror("bind error");exit(1);}
		
	/*步骤3：调用listen启动监听(指定端口监听)
	 * 通知系统去接受来自客户端的连接请求
	 * (将接收到的客户端连接请求放置到对应的队列中)
	 * 第二个参数：指定队列长度为10
	 * */
	if(listen(sockfd, 10)<0)
	{perror("listen error");exit(1);}
	
	/*步骤4：调用accept函数，
	 * 从队列中获得一个客户端的请求连接,
	 * 并返回新的socket描述符。
	 * 若没有客户端连接，调用此函数会阻塞，直到获得一个客户端连接。
	 * */
	struct sockaddr_in clientaddr;
	socklen_t clientaddr_len = sizeof(clientaddr);
	//循环和多个客户端进行连接
	while(1){
		int fd = accept(sockfd,
								(struct sockaddr*)&clientaddr,
								&clientaddr_len);
		if(fd<0){
			perror("accept error");
			continue;
		}
	
		/*步骤5：调用IO函数(read/write)
		 * 和连接的客户端进行双向的通信
		 * */
		out_addr(&clientaddr);
		do_service(fd);
		
		/*步骤6：关闭accept返回的socket*/
		close(fd);
	}
	
	return 0;
}

